package simple.tree;

import struct.TreeNode;

/**
 * 110. 平衡二叉树
 * 给定一个二叉树，判断它是否是高度平衡的二叉树。
 * 本题中，一棵高度平衡二叉树定义为：
 * 一个二叉树每个节点的左右两个子树的高度差的绝对值不超过 1 。
 * 示例 1：
 *   输入：root = [3,9,20,null,null,15,7]
 *   输出：true
 * 示例 2：
 *   输入：root = [1,2,2,3,3,null,null,4,4]
 *   输出：false
 * 示例 3：
 *   输入：root = []
 *   输出：true
 * 提示：
 *   树中的节点数在范围 [0, 5000] 内
 *   -104 <= Node.val <= 104
 * @author 刘学松
 * @date 2021-12-18 14:49
 */
public class 平衡二叉树 {
    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(2);
        root.left.left = new TreeNode(3);
        root.left.right = new TreeNode(3);
        root.left.left.left = new TreeNode(4);
        root.left.left.right = new TreeNode(4);
        System.out.println(isBalanced(root));
    }

    public static boolean isBalanced(TreeNode root) {
//        return root == null || Math.abs(getDeep(root.left) - getDeep(root.right)) < 2 && isBalanced(root.left) && isBalanced(root.right);
        return getDeep(root) >= 0;
    }

    /**
     * 方法1：自顶向下，获取每个节点的最大深度，最后取根节点左右子节点的深度差，时间复杂度 n<sup>2</sup><br/>
     * 方法2：自底向上，获取子节点的最大深度，若左右子节点深度差大于1或者左右子节点不平衡，则返回-1，表示该子树不平衡<br/>
     * 方法1递归中包含重复调用，因此比方法2的时间复杂度更高
     * @param node
     * @return
     */
    public static int getDeep(TreeNode node) {
//        return node == null ? 0 : Math.max(getDeep(node.left), getDeep(node.right)) + 1;
        if (node == null) {
            return 0;
        }
        int leftDeep = getDeep(node.left);
        int rightDeep = getDeep(node.right);
        if (leftDeep == -1 || rightDeep == -1 || Math.abs(leftDeep - rightDeep) > 1) {
            return -1;
        }
        return Math.max(leftDeep, rightDeep) + 1;
    }
}
